package com.xhwl.centiotdata.utils;

import cn.hutool.core.codec.Base64Encoder;
import cn.hutool.http.HttpRequest;
import com.xhwl.common.utils.ChkUtil;
import org.springframework.util.StringUtils;
import sun.misc.BASE64Encoder;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @Author： lianghz
 * @Description：
 * @Date 2021/11/23/ 11:05
 */
public class Base64Utils {

    public static String imageUrlToBase64(String imgUrl) {
        if (StringUtils.isEmpty(imgUrl)) {
            return null;
        }
        URL url = null;
        InputStream is = null;
        ByteArrayOutputStream outStream = null;
        HttpURLConnection httpUrl = null;

        try {
            url = new URL(imgUrl);
            httpUrl = (HttpURLConnection) url.openConnection();
            httpUrl.connect();
            httpUrl.getInputStream();

            is = httpUrl.getInputStream();
            outStream = new ByteArrayOutputStream();

            //创建一个Buffer字符串
            byte[] buffer = new byte[1024];
            //每次读取的字符串长度，如果为-1，代表全部读取完毕
            int len = 0;
            //使用输入流从buffer里把数据读取出来
            while ((len = is.read(buffer)) != -1) {
                //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
                outStream.write(buffer, 0, len);
            }

            // 对字节数组Base64编码
            BASE64Encoder encoder = new BASE64Encoder();
            return encoder.encode(outStream.toByteArray());
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (is != null) {
                    is.close();
                }
                if (outStream != null) {
                    outStream.close();
                }
                if (httpUrl != null) {
                    httpUrl.disconnect();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    public static String getBaseByUrl(String url){
        if(ChkUtil.isNull(url)) {
            return new String();
        }
        byte[] bytes = HttpRequest.get(url).execute().bodyBytes();
        String encode = Base64Encoder.encode(bytes);
        System.out.println(encode);
        return "data:image/jpeg;base64,"+encode;
    }


}
